import os, shutil, sys, re


if __name__ == '__main__':
    dir_name = r"E:\大四下学期\毕业设计\数据\oywh\50Hz单相接地故障模型\gzfl_Two_IIDGs.if12"
    new_dir_A = r"E:\大四下学期\毕业设计\数据\数据\A相接地"
    new_dir_B = r"E:\大四下学期\毕业设计\数据\数据\B相接地"
    new_dir_C = r"E:\大四下学期\毕业设计\数据\数据\C相接地"

    files = os.listdir(dir_name)
    pattern = re.compile(r"^If.*\.inf$")
    for file in files:
        res = re.match(pattern, file).group().split("_")[-1]
        if res:
            # print(11111)
            file_name = r"AB" + res.group().split("_")[-1]
            # shutil.move(os.path.join(dir_name, res.group()), os.path.join(new_dir, file_name))